from codex import *
from time import sleep
import math
import random

start_screen = 'start'
flash_images = 'images'
user_guess = 'guess'
user_fails = 'fails'

mickey = [
    'pics/doggie.jpg',
    'pics/sunDevil.jpg'
]

state = start_screen
points = 0
while True:
    if state == start_screen:
        display.show("Press L to start")
        if buttons.was_pressed(BTN_L):
           state = user_guess
    if state == flash_images:
        num = random.randrange(2)
        poop = mickey[num]
        display.draw_jpg(poop)
        sleep(0.003)
        display.clear()
        looping = True
        while looping:
            if buttons.was_pressed(BTN_A) and poop == mickey[0]:
                points += 1
                state = user_guess
            if buttons.was_pressed(BTN_A) and poop == mickey[1]:
                state = user_fails
            if buttons.was_pressed(BTN_B) and poop == mickey[1]:
                points += 1
                state = user_guess
            if buttons.was_pressed(BTN_B) and poop == mickey[0]:
                state = user_fails
            if buttons.was_pressed(BTN_A) or buttons.was_pressed(BTN_B):
                looping = False
    if state == user_guess:
        display.show('Press A for Pumpkin\nPress B for Christma\ns')
        sleep(2)
        display.clear()
        display.print(3, scale=6)
        sleep(1)
        display.clear()
        display.print(2, scale=6)
        sleep(1)
        display.clear()
        display.print(1, scale=6)
        sleep(1)
        display.clear()
        state = flash_images
        
    if state == user_fails:
        display.show('YOU DIED' + str(points))

#need to fix while looping